# Base image
FROM nvcr.io/nvidia/pytorch:19.02-py3
#RUN apt update -y
#RUN apt upgrade -y

RUN yes | pip install tensorboardX tensorboard tensorflow
